import numpy as np
from matplotlib import pyplot as plt
data = np.array(np.random.uniform(0,1,9)).reshape(3,3)
# print(data)
"""
之前选cmap的参数时用的是：cmap=plt.cmap.RdBu，而现在，我们可以直接用单引号传入参数。
 origin='lower'代表的就是选择的原点的位置。
 而 interpolation 表示画图方式，这里我们使用的是内插法中的 Nearest-neighbor 的方法，其他的方式也都可以随意取选。
"""
plt.imshow(data,interpolation='nearest',cmap='RdBu',origin='lower')
# 增加colorbar
# 们添加一个colorbar ，它可以为我们显示不同颜色的区块所对应的具体数值。
# 其中shrink参数可以用来调整 colorbar 的长度，这里我们使colorbar的长度变短为原来的92%，
# 这样我们2D图像就创建完毕了
plt.colorbar(shrink=0.9)
plt.xticks(())
plt.yticks(())
plt.show()
